package com.autonomous_printer.util;

import org.springframework.stereotype.Component;

@Component
public class FileTypeUtil {

    /**
     * 获取文件类型
     *
     * @param filename
     * @return 文件类型的字符串
     * null表示位置文件类型
     */
    public String getFileType(String filename) {
        String ans = null;
        int index = filename.lastIndexOf('.');
        try {
            ans = filename.substring(index);
        } catch (Exception e) {
            ans = null;
        }
        return ans;
    }

    /**
     * 是否是pdf文件
     * @param filename
     * @return
     */
    public boolean isPDF(String filename) {
        return ".pdf".equals(getFileType(filename));
    }

    /**
     * 是否是word文件
     * @param filename
     * @return
     */
    public boolean isWORD(String filename) {
        return ".doc".equals(getFileType(filename));
    }
}
